导航菜单
首页 >  Macbook cuda环境搭建  > 深度学习开发环境搭建教程(Mac篇)

深度学习开发环境搭建教程(Mac篇)

如果你的Mac不自带NVIDIA的独立显卡(例如15寸以下或者17年新款的Macbook。具体可以在“关于本机->系统报告->图形卡/显示器”里查看),那么你可能无法在这台Mac上使用GPU训练深度学习模型。不过这并不值得遗憾。事实上,我在自己的Macbook上(15-inch,Early 2013,NVIDIA GeForce GT 650M 1024 MB)做了一个简单的测试:在mnist数据集上训练CNN模型时,GPU模式相对于CPU模式仅节省了1/3的时间。这可能要归咎于Mac上赢弱的显卡。相比之下,我更推荐购买一台有着强劲性能的显卡的PC本(比如某些游戏本)来搭建深度学习的开发环境,或者是直接租赁AWS的Instance服务。

下面步入正题。

安装GPU开发环境

几乎所有的主流深度学习框架在使用GPU进行模型训练时都依赖于两个底层环境:CUDA和cuDNN。前者是一个使用GPU进行并行计算的平台,后者是一个封装了使用GPU加速神经网络计算的library。

安装CUDA

确保你的显卡被CUDA所兼容。前往CUDA-capable GPU检查可用的显卡型号。

接下来,安装xcode(通过App Store)和命令行工具:xcode-select --install。这一步会自动安装clang编译器,可以在命令行里输入:/usr/bin/cc --version进行验证。clang编译器也是安装Theano的预备环境之一。

前往官网下载并安装CUDA8.0(dmg安装包)。注意选择相应的平台。这一步会同时安装CUDA driver和CUDA toolkit。

配置环境变量。打开~/.bash_profile,输入四行命令:

export CUDA_ROOT=/Developer/NVIDIA/CUDA-8.0export PATH=$CUDA_ROOT/bin${PATH:+:${PATH}}export DYLD_LIBRARY_PATH=$CUDA_ROOT/lib${DYLD_LIBRARY_PATH:+:${DYLD_LIBRARY_PATH}}export LD_LIBRARY_PATH=$CUDA_ROOT/lib:$LD_LIBRARY_PATH

其中,环境变量CUDA_ROOT给Theano指定了CUDA的安装目录,也可以被设置为/usr/local/cuda(这两个路径下的文件均通过软

相关推荐: